package mysql

import (
	"cloud-storage/global"
)

type FileStore struct {
	Id          int
	UserId      int
	CurrentSize int64
	MaxSize     int64
}

// CapacityIsEnough 判断用户容量是否足够
func CapacityIsEnough(fileSize int64, fileStoreId int) bool {
	var fileStore FileStore
	global.DB.Where("id = ?", fileStoreId).First(&fileStore)
	if fileStore.MaxSize-(fileSize/1024) < 0 {
		return false
	}

	return true
}

func GetUserFileStore(userId int) (fileStore FileStore, err error) {
	err = global.DB.Where("user_id = ?", userId).First(&fileStore).Error
	return
}
